
package mvcc

import (
	"go.etcd.io/etcd/api/v3/mvccpb"
	"go.etcd.io/etcd/pkg/v3/traceutil"
)

// 在 watchableStore . Write （） 方法中返回 的是 watchableStoreTxnWrite 实例， watchableStoreTxnWrite
// 中封装了storeTxnWrite 和关联的 watchableStore 实例。 watchableStoreTxnWrite
// 重写 了 End （） 方法， 如果当 前读写事务中存在更新操作，则会在 End（） 方法中触发相应的 watcher
// 实例 ，
func (tw *watchableStoreTxnWrite) End() {
	changes := tw.Changes()	// 当前读写事务中执行的 更新操作
	if len(changes) == 0 {	// 如果当前事务 中 没有任何更新操作 ， 则直接结束当前事务即可
		tw.TxnWrite.End()
		return
	}

	rev := tw.Rev() + 1
	evs := make([]mvccpb.Event, len(changes))
	for i, change := range changes {	// 将当前事务中的更新操作转换成对应的 Event 事件
		evs[i].Kv = &changes[i]
		if change.CreateRevision == 0 {
			evs[i].Type = mvccpb.DELETE
			evs[i].Kv.ModRevision = rev
		} else {
			evs[i].Type = mvccpb.PUT
		}
	}

	// 在可监视存储锁定下结束写入txn，以便更新可见
	// 异步事件发布检查当前存储版本
	tw.s.mu.Lock()
	tw.s.notify(rev, evs)	// 调用 watchableStore.notify（）方法，将上述 Event 事件发送出 去
	tw.TxnWrite.End()	// 结束当前读写事务
	tw.s.mu.Unlock()
}

type watchableStoreTxnWrite struct {
	TxnWrite
	s *watchableStore
}

func (s *watchableStore) Write(trace *traceutil.Trace) TxnWrite {
	return &watchableStoreTxnWrite{s.store.Write(trace), s}
}
